../../../../../../../../models/demos/metal_BERT_large_11/tests/test_bert_batch_dram.py